<?php

// +----------------------------------------------------------------------

// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]

// +----------------------------------------------------------------------

// | Copyright (c) 2006-2013 http://thinkphp.cn All rights reserved.

// +----------------------------------------------------------------------

// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )

// +----------------------------------------------------------------------

// | Author: 麦当苗儿 <zuojiazi@vip.qq.com> <http://www.zjzit.cn>

// +----------------------------------------------------------------------



namespace Think\Upload\Driver;

class Ftp{

    /**

     * 上传文件根目录

     * @var string

     */

    private $rootPath;



    /**

     * 本地上传错误信息

     * @var string

     */

    private $error = ''; //上传错误信息



    /**

     * FTP连接

     * @var resource

     */

    private $link;



    private $config = array(

        'host'     => '', //服务器

        'port'     => 21, //端口

        'timeout'  => 90, //超时时间

        'username' => '', //用户名

        'password' => '', //密码

    );



    /**

     * 构造函数，用于设置上传根路径

     * @param string $root   根目录

     * @param array  $config FTP配置

     */

	public function __construct($root, $config){

        /* 默认FTP配置 */

        $this->config = array_merge($this->config, $config);



        /* 登录FTP服务器 */

        if(!$this->login()){

            throw new \Exception($this->error);

        }

        

        /* 设置根目录 */

        $this->rootPath = ftp_pwd($this->link) . '/' . ltrim($root, '/');

	}



    /**

     * 检测上传根目录

     * @return boolean true-检测通过，false-检测失败

     */

    public function checkRootPath(){

        if(!@ftp_chdir($this->link, $this->rootPath)){

            $this->error = '上传根目录不存在！';

            return false;

        }

        return true;

    }



    /**

     * 检测上传目录

     * @param  string $savepath 上传目录

     * @return boolean          检测结果，true-通过，false-失败

     */

	public function checkSavePath($savepath){

        /* 检测并创建目录 */

    	if (!$this->mkdir($savepath)) {

    		return false;

    	} else {

            //TODO:检测目录是否可写

    		return true;

    	}

    }



    /**

     * 保存指定文件

     * @param  array   $file    保存的文件信息

     * @param  boolean $replace 同名文件是否覆盖

     * @return boolean          保存状态，true-成功，false-失败

     */

    public function save($file, $replace=true) {

        $filename = $this->rootPath . $file['savepath'] . $file['savename'];



        /* 不覆盖同名文件 */

        // if (!$replace && is_file($filename)) {

        //     $this->error = '存在同名文件' . $file['savename'];

        //     return false;

        // }



        /* 移动文件 */

        if (!ftp_put($this->link, $filename, $file['tmp_name'], FTP_BINARY)) {

            $this->error = '文件上传保存错误！';

            return false;

        }

        

        return true;

    }



    /**

     * 创建目录

     * @param  string $savepath 要创建的穆里

     * @return boolean          创建状态，true-成功，false-失败

     */

    public function mkdir($savepath){

        $dir = $this->rootPath . $savepath;

        if(ftp_chdir($this->link, $dir)){

            return true;

        }



        if(ftp_mkdir($this->link, $dir)){

            return true;

        } elseif($this->mkdir(dirname($savepath)) && ftp_mkdir($this->link, $dir)) {

            return true;

        } else {

            $this->error = "目录 {$savepath} 创建失败！";

            return false;

        }

    }



    /**

     * 获取最后一次上传错误信息

     * @return string 错误信息

     */

    public function getError(){

        return $this->error;

    }



    /**

     * 登录到FTP服务器

     * @return boolean true-登录成功，false-登录失败

     */

    private function login(){

        extract($this->config);

        $this->link = ftp_connect($host, $port, $timeout);

        if($this->link) {

            if (ftp_login($this->link, $username, $password)) {

               return true;

            } else {

                $this->error = "无法登录到FTP服务器：username - {$username}";

            }

        } else {

            $this->error = "无法连接到FTP服务器：{$host}";

        }

        return false;

    }



    /**

     * 析构方法，用于断开当前FTP连接

     */

    public function __destruct() {

        ftp_close($this->link);

    }



}

